﻿#pragma once

# if defined(TXR_ALGO_DLM_CLS_V2_LIB)
#  define TXR_ALGO_DLM_CLS_V2_EXPORT Q_DECL_EXPORT
# else
#  define TXR_ALGO_DLM_CLS_V2_EXPORT Q_DECL_IMPORT
# endif
#include "../txr_algo_dlm_cls/txr_algo_dlm_cls_impl.h"

namespace txr_algo_dlm_cls
{
	class InferClsV2;

	class TxrAlgoDlmClsV2 : public TxrAlgoDlmClsImpl
	{
	public:
		TxrAlgoDlmClsV2();
		virtual ~TxrAlgoDlmClsV2();

		tk_bool Load(st_dlm_cfg& cfg) override;
		void Release();
		tk_int32 BatchSize() override;
		void Detect(st_detect_unit* p_unit, tk_int32 size) override;
	private:
		InferClsV2* m_p_infer;
	};
}//namespace txr_algo_dlm_cls
